BERT_BASE_DIR=bert-base-chinese
DATA_DIR=pku_ner/
OUTPUT_DIR=model
python ner.py \
    --model_name_or_path ${BERT_BASE_DIR} \
    --do_train True \
    --do_eval True \
    --do_test True \
    --max_seq_length 256 \
    --train_file ${DATA_DIR}/train_sen.pkl \
    --train_label ${DATA_DIR}/train_lab.pkl \
    --eval_file ${DATA_DIR}/valid_sen.pkl \
    --eval_label ${DATA_DIR}/valid_lab.pkl \
    --test_file ${DATA_DIR}/test_sen.pkl \
    --test_label ${DATA_DIR}/test_lab.pkl \
    --train_batch_size 32 \
    --eval_batch_size 32 \
    --num_train_epochs 2 \
    --do_lower_case \
    --logging_steps 200 \
    --need_birnn True \
    --rnn_dim 256 \
    --clean True \
    --output_dir $OUTPUT_DIR